Machine learning-based predictive model